package cn.digitalize.system.core.util;

import cn.dev33.satoken.stp.StpUtil;
import cn.digitalize.system.modular.auth.result.BaseLoginUser;

/**
 * @author Created by ZhangCheng on 2023/06/05 (用户登录工具类)
 */
public class StpLoginUserUtil {

    private static final String TOKEN_SESSION_KEY = "loginUser";

    /**
     * 设置当前登录用户
     *
     * @return
     */
    public static void setLoginUser(BaseLoginUser baseLoginUser) {
        // 缓存用户信息，此处使用TokenSession为了指定时间内无操作则自动下线
        StpUtil.getTokenSession().set(TOKEN_SESSION_KEY, baseLoginUser);
    }
    /**
     * 获取当前登录用户
     *
     * @return
     */
    public static BaseLoginUser getLoginUser() {
        return (BaseLoginUser) StpUtil.getTokenSession().get(TOKEN_SESSION_KEY);
    }
}
